package com.yzh.openai.domain.openai.model.entity;

import com.yzh.openai.domain.openai.model.valobj.UserAccountStatusVO;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.util.Arrays;
import java.util.List;

/**
 * 用户账户实体类对象
 */
@Data
@NoArgsConstructor
@AllArgsConstructor
@Builder
public class UserAccountEntity {
    /**
     * 用户ID，这里使用微信openid作为唯一ID
     */
    private String openid;

    /**
     * 总量额度：分配的总使用次数
     */
    private Integer totalQuota;

    /**
     * 剩余额度：剩余可使用次数
     */
    private Integer surplusQuota;

    /**
     * 可用模型
     */
    private List<String> allowModelTypeList;

    /**
     * 账户状态：0-可用、1-冻结
     */
    private UserAccountStatusVO status;
    
    public void genModelTypes(String modelTypes) {
        String[] split = modelTypes.split(",");
        this.allowModelTypeList = Arrays.asList(split);
    }
}
